import BotInit


def start():
    botObj = BotInit.wxBot()
    while True:
        print("\n微信朋友管理与图文交互系统\n")
        print("功能列表\n 1.导出好友信息\n 2.查看聊天组\n 3.选择聊天对象\n 4.创建好友城市分布饼图\n 5.退出\n")
        num = int(input("请选择：\n"))
        if num == 1:
            botObj.save_xsl()
        elif num == 2:
            botObj.watch_chat()
        elif num == 3:
            botObj.talk()
        elif num == 4:
            botObj.create_html()
        elif num == 5:
            botObj.logout()
            break
        else:
            print("无此命令,请重新选择！")


if __name__ == '__main__':
    start()
